TensorRT Inference API

Detailed Description

The GIE_Context class provides interfaces to load the Caffe model and perform inference with the TensorRT model.

Data Structures
class	GIE_Context

Macros
#define	GOOGLENET_SINGLE_CLASS   0
#define	GOOGLENET_THREE_CLASS   1
#define	HELNET_THREE_CLASS   2

Variables
struct {
const int   classCnt
float   THRESHOLD [3]
const char *   INPUT_BLOB_NAME
const char *   OUTPUT_BLOB_NAME
const char *   OUTPUT_BBOX_NAME
const int   STRIDE
const int   WORKSPACE_SIZE
float   bbox_output_scales [4]
}	g_pModelNetAttr
struct {
const int   classCnt
float   THRESHOLD [3]
const char *   INPUT_BLOB_NAME
const char *   OUTPUT_BLOB_NAME
const char *   OUTPUT_BBOX_NAME
const int   STRIDE
const int   WORKSPACE_SIZE
float   bbox_output_scales [4]
}	gModelNetAttr [3]

Macro Definition Documentation

#define GOOGLENET_SINGLE_CLASS   0

Definition at line 64 of file gie_inference.h.

#define GOOGLENET_THREE_CLASS   1

Definition at line 65 of file gie_inference.h.

#define HELNET_THREE_CLASS   2

Definition at line 66 of file gie_inference.h.

Variable Documentation

float { ... } bbox_output_scales[4]

Definition at line 76 of file gie_inference.h.

const { ... } classCnt

Definition at line 69 of file gie_inference.h.

struct { ... } * g_pModelNetAttr

struct { ... } gModelNetAttr[3]

const { ... } INPUT_BLOB_NAME

Definition at line 71 of file gie_inference.h.

const { ... } OUTPUT_BBOX_NAME

Definition at line 73 of file gie_inference.h.

const { ... } OUTPUT_BLOB_NAME

Definition at line 72 of file gie_inference.h.

const { ... } STRIDE

Definition at line 74 of file gie_inference.h.

float { ... } THRESHOLD[3]

Definition at line 70 of file gie_inference.h.

const { ... } WORKSPACE_SIZE

Definition at line 75 of file gie_inference.h.